package com.lidong.filter;

import javax.servlet.http.HttpServletRequest;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;

import com.netflix.zuul.ZuulFilter;
import com.netflix.zuul.context.RequestContext;

/**
 * @Description : zuul不仅只是路由，并且还能过滤，做一些安全验证
 * @Author : lidong
 * @Date : 2017年11月9日
 */
@Component
public class MyFilter extends ZuulFilter {

	private static Logger logger = LoggerFactory.getLogger(MyFilter.class);

	/**
	 * shouldFilter：这里可以写逻辑判断，是否要过滤，本文true,永远过滤
	 */
	@Override
	public boolean shouldFilter() {
		
		logger.info("MyFilter.shouldFilter()");
		
		return true;
	}

	/**
	 * run：过滤器的具体逻辑。可用很复杂，包括查sql，nosql去判断该请求到底有没有权限访问
	 */
	@Override
	public Object run() {
		
		logger.info("MyFilter.run()");
		
		RequestContext ctx = RequestContext.getCurrentContext();
		HttpServletRequest request = ctx.getRequest();
		
		logger.info(String.format("%s >>> %s", request.getMethod(), request.getRequestURL().toString()));
		
		Object accessToken = request.getParameter("token");
		if (accessToken == null) {
			logger.warn("token is empty!!!");
			ctx.setSendZuulResponse(false);
			ctx.setResponseStatusCode(401);
			try {
				ctx.getResponse().getWriter().write("token is empty!!!");
			} catch (Exception e) {
				logger.error(e.getMessage(), e);
			}

			return null;
		}
		logger.info("ok");

		return null;
	}

	/**
	 * filterType：返回一个字符串代表过滤器的类型，在zuul中定义了四种不同生命周期的过滤器类型，具体如下：
	 * pre：路由之前
	 * routing：路由之时 
	 * post： 路由之后 
	 * error：发送错误调用 
	 */
	@Override
	public String filterType() {
		
		logger.info("MyFilter.filterType()");
		
		return "pre";
	}

	/**
	 * filterOrder：过滤的顺序
	 */
	@Override
	public int filterOrder() {
		
		logger.info("MyFilter.filterOrder()");
		
		return 0;
	}

}
